# 生产环境尽量选择 Large 的模型
# 导入llamaIndex的embedding包  pip install llama-index-embeddings-huggingface
from llama_index.embeddings.huggingface import HuggingFaceEmbedding


# 创建本地模型  其中模型名称为：BAAI/bge-small-zh-v1.5(从hugging face中下载)，缓存路径为：模型保存的路径
def embed_model_local_bge_small(**kwargs):
    return HuggingFaceEmbedding(model_name="BAAI/bge-small-zh-v1.5",
                                cache_folder="/Volumes/work/embedding/bge-small-zh-v1.5",
                                **kwargs)